剑指Offer 7 用两个栈实现队列

题目描述

用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。

思路

栈是先进后出的,那么两个栈,一个叫做原始栈,push都到这里,另一个叫做队列栈,我们pop的时候,将原始栈中pop出的元素都push到队列栈中;整体顺序就调换了;
如果队列栈还有元素,那我们就直接pop队列栈;因为队列栈实际上就是队列了;

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
public static class Solution {
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
public void push(int node) {
// while (stack2.size()!=0) //上次pop时,栈2可能还有残留元素,首先想到了放回栈1
// {
// stack1.push(stack2.pop());
// }
stack1.push(node);
}
public int pop() {
if (stack2.isEmpty()) { //如果栈2是空的,就直接pop栈2,否则从栈一获取元素
while (stack1.size() != 0) {
stack2.push(stack1.pop());
}
stack1.empty(); //放到栈2后,栈2就是正确排列的队列了,栈一就可以清空了
}
return stack2.pop();
}
}

收获

  1. 有想法就好啊;一看到题目其实我的思路就出来了,但是还是没想全面,画图还是必须的,最好一步一步走下来,再开始编程;